// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <string>

#include "base/command_line.h"
#include "base/macros.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/material_design/material_design_controller.h"
#include "ui/base/test/material_design_controller_test_api.h"
#include "ui/base/ui_base_switches.h"

namespace ui {
namespace {

    // Test fixture for the MaterialDesignController class.
    class MaterialDesignControllerTest : public testing::Test {
    public:
        MaterialDesignControllerTest();
        ~MaterialDesignControllerTest() override;

        // testing::Test:
        void SetUp() override;

    private:
        DISALLOW_COPY_AND_ASSIGN(MaterialDesignControllerTest);
    };

    MaterialDesignControllerTest::MaterialDesignControllerTest()
    {
    }

    MaterialDesignControllerTest::~MaterialDesignControllerTest()
    {
    }

    void MaterialDesignControllerTest::SetUp()
    {
        testing::Test::SetUp();

        // Ensure other tests aren't polluted by a Mode set in these tests. The Mode
        // has to be cleared in SetUp and not in TearDown because when the test suite
        // starts up it triggers a call to ResourceBundle::LoadCommonResources()
        // which calls MaterialDesignController::IsModeMaterial(), thereby
        // initializing the mode and potentially causing the first test to fail.
        test::MaterialDesignControllerTestAPI::UninitializeMode();
    }

#if !defined(ENABLE_TOPCHROME_MD)

    // Verify the Mode maps to Mode::NON_MATERIAL when the compile time flag is not
    // defined.
    TEST_F(MaterialDesignControllerTest,
        NonMaterialModeWhenCompileTimeFlagDisabled)
    {
        EXPECT_EQ(MaterialDesignController::Mode::NON_MATERIAL,
            MaterialDesignController::GetMode());
    }

#else

    // Verify command line value "material" maps to Mode::MATERIAL when the compile
    // time flag is defined.
    TEST_F(MaterialDesignControllerTest,
        EnabledCommandLineValueMapsToMaterialModeWhenCompileTimeFlagEnabled)
    {
        base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
            switches::kTopChromeMD, "material");
        EXPECT_EQ(MaterialDesignController::Mode::MATERIAL_NORMAL,
            MaterialDesignController::GetMode());
    }

    // Verify command line value "material-hybrid" maps to Mode::MATERIAL_HYBRID
    // when the compile time flag is defined.
    TEST_F(
        MaterialDesignControllerTest,
        EnabledHybridCommandLineValueMapsToMaterialHybridModeWhenCompileTimeFlagEnabled)
    {
        base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
            switches::kTopChromeMD, "material-hybrid");
        EXPECT_EQ(MaterialDesignController::Mode::MATERIAL_HYBRID,
            MaterialDesignController::GetMode());
    }

    // Verify command line value "" maps to the default mode when the compile time
    // flag is defined.
    TEST_F(
        MaterialDesignControllerTest,
        DisabledCommandLineValueMapsToNonMaterialModeWhenCompileTimeFlagEnabled)
    {
        base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
            switches::kTopChromeMD, "");
        EXPECT_EQ(MaterialDesignController::DefaultMode(),
            MaterialDesignController::GetMode());
    }

    // Verify the current mode is reported as the default mode when no command line
    // flag is defined.
    TEST_F(MaterialDesignControllerTest,
        NoCommandLineValueMapsToNonMaterialModeWhenCompileTimeFlagEnabled)
    {
        ASSERT_FALSE(base::CommandLine::ForCurrentProcess()->HasSwitch(
            switches::kTopChromeMD));
        EXPECT_EQ(MaterialDesignController::DefaultMode(),
            MaterialDesignController::GetMode());
    }

    // Verify an invalid command line value uses the default mode.
    TEST_F(MaterialDesignControllerTest, InvalidCommandLineValue)
    {
        const std::string kInvalidValue = "1nvalid-valu3";
        base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
            switches::kTopChromeMD, kInvalidValue);
        EXPECT_EQ(MaterialDesignController::DefaultMode(),
            MaterialDesignController::GetMode());
    }

    // Verify that MaterialDesignController::IsModeMaterial() will initialize the
    // mode if it hasn't been initialized yet.
    TEST_F(MaterialDesignControllerTest, IsModeMaterialInitializesMode)
    {
        base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
            switches::kTopChromeMD, "material");
        EXPECT_TRUE(MaterialDesignController::IsModeMaterial());
    }

#endif // !defined(ENABLE_TOPCHROME_MD)

} // namespace
} // namespace ui
